from selenium.webdriver.common.by import By

from Hogwarts.Homeworks.hw0627.page_objects.base_page import BasePage


"""
PAGE：企业微信首页
"""


class MainPage(BasePage):
    # OBJECTS
    _CONTACT = (By.ID, "menu_contacts")

    def goto_contact(self):
        from Hogwarts.Homeworks.hw0627.page_objects.contact_page import ContactPage

        self.find_and_click(*self._CONTACT)
        return ContactPage(self.driver)
